list comprehensions hackerrank solution